package com.ruoyi.framework.security.handle;

import java.io.IOException;
import java.io.Serializable;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.ruoyi.common.exception.user.OAAuthenticationException;
import org.springframework.security.authentication.*;
import org.springframework.security.core.AuthenticationException;
import org.springframework.security.web.AuthenticationEntryPoint;
import org.springframework.stereotype.Component;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.common.constant.HttpStatus;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.utils.ServletUtils;
import com.ruoyi.common.utils.StringUtils;

/**
 * 认证失败处理类 返回未授权
 * 
 * @author ruoyi
 */
@Component
public class AuthenticationEntryPointImpl implements AuthenticationEntryPoint, Serializable
{
    private static final long serialVersionUID = -8970718410437077606L;

    @Override
    public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException e)
            throws IOException
    {
        int code = HttpStatus.UNAUTHORIZED;
        String msg;

        // 根据不同的异常类型返回不同的错误信息
        if (e instanceof OAAuthenticationException) {
            msg = StringUtils.format("请求访问：{}，OA认证失败，{}，无法访问系统资源",
                    request.getRequestURI(), e.getMessage());
        } else if (e instanceof BadCredentialsException) {
            msg = StringUtils.format("请求访问：{}，用户名或密码错误，无法访问系统资源",
                    request.getRequestURI());
        } else if (e instanceof DisabledException) {
            msg = StringUtils.format("请求访问：{}，账户已被禁用，无法访问系统资源",
                    request.getRequestURI());
            code = HttpStatus.FORBIDDEN;
        } else if (e instanceof LockedException) {
            msg = StringUtils.format("请求访问：{}，账户已被锁定，无法访问系统资源",
                    request.getRequestURI());
            code = HttpStatus.FORBIDDEN;
        } else if (e instanceof AccountExpiredException) {
            msg = StringUtils.format("请求访问：{}，账户已过期，无法访问系统资源",
                    request.getRequestURI());
            code = HttpStatus.FORBIDDEN;
        } else if (e instanceof CredentialsExpiredException) {
            msg = StringUtils.format("请求访问：{}，凭证已过期，无法访问系统资源",
                    request.getRequestURI());
        } else {
            msg = StringUtils.format("请求访问：{}，认证失败，无法访问系统资源",
                    request.getRequestURI());
        }

        ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(code, msg)));
    }
}
